import pymysql
def connect_to_database():
    conn = pymysql.connect(user='root', password='544002', host='localhost', database='shiyu', port=3307)
    cursor = conn.cursor()
    return conn, cursor
